/*
有一幅以 m x n 的二维整数数组表示的图画 image ，其中 image[i][j] 表示该图画的像素值大小。

你也被给予三个整数 sr ,  sc 和 newColor 。你应该从像素 image[sr][sc] 开始对图像进行 上色填充 。

为了完成 上色工作 ，从初始像素开始，记录初始坐标的 上下左右四个方向上 像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应 四个方向上 像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为 newColor 。

最后返回 经过上色渲染后的图像 。

输入: image = [[1,1,1],[1,1,0],[1,0,1]]，sr = 1, sc = 1, newColor = 2
输出: [[2,2,2],[2,2,0],[2,0,1]]
解析: 在图像的正中间，(坐标(sr,sc)=(1,1)),在路径上所有符合条件的像素点的颜色都被更改成2。
注意，右下角的像素没有更改为2，因为它不是在上下左右四个方向上与初始点相连的像素点。
示例 2:

输入: image = [[0,0,0],[0,0,0]], sr = 0, sc = 0, newColor = 2
输出: [[2,2,2],[2,2,2]]
*/
package main

import "fmt"

func main(){
	image := [][]int{{1,1,1},{1,1,0},{1,0,1}}
	sr := 1
	sc := 1
	newColor := 2

	newImage := floodFill(image, sr, sc, newColor)
	fmt.Println(newImage)
}

//BFS
// type BfsStruct struct{
// 	x int
// 	y int
// 	color int
// }

// func floodFill(image [][]int, sr int, sc int, color int) [][]int {
// 	var q []*BfsStruct
// 	if len(image) <= 0 || len(image[0]) <= 0 ||sr > len(image) -1 || sc > len(image[0]){
// 		return image
// 	}
// 	q = append(q, &BfsStruct{x: sr, y: sc, color: image[sr][sc]})
// 	oc := image[sr][sc]
// 	if oc == color{
// 		return image
// 	}
// 	for len(q) > 0{
// 		point := q[0]
// 		q = q[1:]
// 		if point.color != color{
// 			image[point.x][point.y] = color
// 		}
// 		if point.x - 1 >= 0 && oc == image[point.x - 1][point.y]{
// 			q = append(q, &BfsStruct{x: point.x - 1, y: point.y, color: image[point.x - 1][point.y]})
// 		}
// 		if point.x + 1 < len(image) && oc == image[point.x + 1][point.y]{
// 			q = append(q, &BfsStruct{x: point.x + 1, y: point.y, color: image[point.x + 1][point.y]})
// 		}
// 		if point.y - 1 >= 0 && oc == image[point.x][point.y - 1]{
// 			q = append(q, &BfsStruct{x: point.x , y: point.y - 1, color: image[point.x][point.y - 1]})
// 		}
// 		if point.y + 1 < len(image[0]) && oc == image[point.x][point.y + 1]{
// 			q = append(q, &BfsStruct{x: point.x , y: point.y + 1, color: image[point.x][point.y + 1]})
// 		}
// 	}
// 	return image
// }


//DFS
func floodFill(image [][]int, sr int, sc int, color int) [][]int {
	if image[sr][sc] == color{
		return image
	}

	dfs(image, sr,sc,image[sr][sc],color)
	return image
}

func dfs(image [][]int, sr,sc int, oc,nc int){
	if sr < 0 || sc < 0 || sr >= len(image) || sc >= len(image[0]) || image[sr][sc] != oc{
		return
	}
	image[sr][sc] = nc

	dfs(image, sr - 1, sc, oc, nc)
	dfs(image, sr + 1, sc, oc, nc )
	dfs(image, sr, sc - 1, oc, nc)
	dfs(image, sr, sc + 1, oc, nc )

}
